package com.zhht.htzx.utils;

import android.content.ContentValues;
import android.content.Context;
import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.graphics.Matrix;
import android.graphics.Paint;
import android.net.Uri;
import android.os.Build;
import android.os.Environment;
import android.provider.MediaStore;
import android.util.Log;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;

public class SharePosterUtils {
    private static final String TAG = "SharePosterUtils";
    private static final int QR_CODE_SIZE = 300; // 二维码大小
    private static final int QR_CODE_MARGIN = 0; // 二维码边距设为0，去掉白边

    /**
     * 生成二维码（无白边）
     * @param content 二维码内容
     * @return 二维码Bitmap
     */
    public static Bitmap generateQRCode(String content) {
        try {
            Map<EncodeHintType, Object> hints = new HashMap<>();
            hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
            hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
            hints.put(EncodeHintType.MARGIN, QR_CODE_MARGIN);

            BitMatrix bitMatrix = new MultiFormatWriter().encode(
                    content,
                    BarcodeFormat.QR_CODE,
                    QR_CODE_SIZE,
                    QR_CODE_SIZE,
                    hints
            );

            int width = bitMatrix.getWidth();
            int height = bitMatrix.getHeight();
            Bitmap bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);

            for (int x = 0; x < width; x++) {
                for (int y = 0; y < height; y++) {
                    bitmap.setPixel(x, y, bitMatrix.get(x, y) ? 0xFF000000 : 0x00FFFFFF);
                }
            }
            return bitmap;
        } catch (WriterException e) {
            Log.e(TAG, "Generate QR code failed", e);
            return null;
        }
    }

    /**
     * 将二维码直接绘制到海报上
     * @param posterBitmap 海报图片
     * @param qrCodeBitmap 二维码图片
     * @param x 二维码在海报上的x坐标
     * @param y 二维码在海报上的y坐标
     * @return 绘制了二维码的海报
     */
    public static Bitmap composePoster(Bitmap posterBitmap, Bitmap qrCodeBitmap, int x, int y) {
        if (posterBitmap == null || qrCodeBitmap == null) {
            return null;
        }

        // 创建海报的副本，避免修改原图
        Bitmap result = posterBitmap.copy(Bitmap.Config.ARGB_8888, true);

        Canvas canvas = new Canvas(result);
        Paint paint = new Paint();
        paint.setAntiAlias(true);

        // 直接在海报上绘制二维码
        canvas.drawBitmap(qrCodeBitmap, x, y, paint);

        return result;
    }

    /**
     * 直接在原海报上绘制二维码（修改原图）
     * @param posterBitmap 海报图片（会被修改）
     * @param qrCodeBitmap 二维码图片
     * @param x 二维码在海报上的x坐标
     * @param y 二维码在海报上的y坐标
     * @return 是否绘制成功
     */
    public static boolean drawQRCodeOnPoster(Bitmap posterBitmap, Bitmap qrCodeBitmap, int x, int y) {
        if (posterBitmap == null || qrCodeBitmap == null) {
            return false;
        }

        // 确保海报是可变的
        if (!posterBitmap.isMutable()) {
            return false;
        }

        Canvas canvas = new Canvas(posterBitmap);
        Paint paint = new Paint();
        paint.setAntiAlias(true);

        // 直接在海报上绘制二维码
        canvas.drawBitmap(qrCodeBitmap, x, y, paint);

        return true;
    }

    /**
     * 保存图片到本地应用目录
     * @param context 上下文
     * @param bitmap 要保存的图片
     * @param fileName 文件名
     * @return 保存后的文件Uri
     */
    public static Uri saveImageToGallery(Context context, Bitmap bitmap, String fileName) {
        if (bitmap == null) {
            return null;
        }

        File imagesDir = new File(context.getExternalFilesDir(Environment.DIRECTORY_PICTURES), "SharePoster");
        if (!imagesDir.exists()) {
            imagesDir.mkdirs();
        }

        File imageFile = new File(imagesDir, fileName + ".png");
        try {
            FileOutputStream fos = new FileOutputStream(imageFile);
            bitmap.compress(Bitmap.CompressFormat.PNG, 100, fos);
            fos.flush();
            fos.close();
            return Uri.fromFile(imageFile);
        } catch (IOException e) {
            Log.e(TAG, "Save image failed", e);
            return null;
        }
    }

    /**
     * 保存图片到系统相册（Android 10及以上版本）
     * @param context 上下文
     * @param bitmap 要保存的图片
     * @param fileName 文件名
     * @return 保存后的文件Uri
     */
    public static Uri saveImageToSystemGallery(Context context, Bitmap bitmap, String fileName) {
        if (bitmap == null) {
            return null;
        }

        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) {
            // Android 10及以上使用MediaStore API
            ContentValues values = new ContentValues();
            values.put(MediaStore.Images.Media.DISPLAY_NAME, fileName + ".png");
            values.put(MediaStore.Images.Media.MIME_TYPE, "image/png");
            values.put(MediaStore.Images.Media.RELATIVE_PATH, Environment.DIRECTORY_PICTURES + "/SharePoster");

            Uri uri = context.getContentResolver().insert(MediaStore.Images.Media.EXTERNAL_CONTENT_URI, values);
            if (uri != null) {
                try (OutputStream os = context.getContentResolver().openOutputStream(uri)) {
                    if (os != null) {
                        bitmap.compress(Bitmap.CompressFormat.PNG, 100, os);
                        return uri;
                    }
                } catch (IOException e) {
                    Log.e(TAG, "Save image to system gallery failed", e);
                }
            }
        } else {
            // Android 9及以下版本
            File imagesDir = new File(Environment.getExternalStoragePublicDirectory(Environment.DIRECTORY_PICTURES), "SharePoster");
            if (!imagesDir.exists()) {
                imagesDir.mkdirs();
            }

            File imageFile = new File(imagesDir, fileName + ".png");
            try {
                FileOutputStream fos = new FileOutputStream(imageFile);
                bitmap.compress(Bitmap.CompressFormat.PNG, 100, fos);
                fos.flush();
                fos.close();
                
                // 通知系统相册更新
                context.sendBroadcast(new android.content.Intent(android.content.Intent.ACTION_MEDIA_SCANNER_SCAN_FILE, Uri.fromFile(imageFile)));
                return Uri.fromFile(imageFile);
            } catch (IOException e) {
                Log.e(TAG, "Save image to system gallery failed", e);
            }
        }
        return null;
    }

    /**
     * 缩放二维码到指定大小
     * @param qrCodeBitmap 原始二维码
     * @param targetSize 目标大小
     * @return 缩放后的二维码
     */
    public static Bitmap scaleQRCode(Bitmap qrCodeBitmap, int targetSize) {
        if (qrCodeBitmap == null) {
            return null;
        }

        float scale = (float) targetSize / qrCodeBitmap.getWidth();
        Matrix matrix = new Matrix();
        matrix.postScale(scale, scale);

        return Bitmap.createBitmap(
                qrCodeBitmap,
                0,
                0,
                qrCodeBitmap.getWidth(),
                qrCodeBitmap.getHeight(),
                matrix,
                true
        );
    }
} 